AI-based AR training system